import com.erp.erp_entitys.Menum.MenumEntity;
import com.erp.erp_entitys.Resource.ResourceEntity;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @ClassName menumtest
 * @Description TODO
 * @Author songjizhen
 * @Date 2020-07-13 9:31
 * @Version 1.0
 */
public class menumtest {
    public static void main(String[] args) {
        MenumEntity me0=new MenumEntity();
        MenumEntity me1=new MenumEntity();
        MenumEntity me2=new MenumEntity();
        me2.setId(2);
        me2.setCode("2");
        me2.setPid("1");
        me1.setCode("1");
        me1.setId(1);
        me1.setPid("0");
        List<MenumEntity> list1=new ArrayList<MenumEntity>();
        list1.add(me2);
       // me1.setList(list1);
        me0.setId(0);
        me0.setCode("0");
        List<MenumEntity> list0=new ArrayList<MenumEntity>();
        list0.add(me1);
       // me0.setList(list0);
        Map<String, MenumEntity> map=new HashMap<String, MenumEntity>();
        map.put("2",me2);
        map.put("1",me1);
        map.put("0",me0);
        menumtest m=new menumtest();
        MenumEntity parentTree = m.getParentTree(me2, map);
    }
    private MenumEntity getParentTree(MenumEntity m, Map<String, MenumEntity> map){
        String pid=m.getPid();
        MenumEntity me=new MenumEntity();
        if(map.get(pid).getCode() !="0"){
            List<MenumEntity> list=new ArrayList<MenumEntity>();
            list.add(m);
            map.get(pid).setList(list);
            me= this.getParentTree(map.get(pid), map);

        }else{
            me=map.get(pid);
            List<MenumEntity> list=new ArrayList<MenumEntity>();
            list.add(m);
            me.setList(list);
        }



        //
        return me;
    }
}
